DATA_PATH = "data/raw_data.txt"
PREPROCESSED_PATH = "data/preprocessed_data.pkl"
EMBEDDINGS_PATH = "embeddings/knowledge_embeddings.pkl"
INDEX_PATH = "faiss/knowledge_index.index"
# INDEX_PATH = "index/faiss_index.index"

DEVICE = "cpu"  # 强制使用CPU，更稳定
MAX_INPUT_LENGTH = 256   # 减小输入长度
MAX_OUTPUT_LENGTH = 64   # 减小输出长度
TOP_K_RESULTS = 3
MODEL_NAME = "t5-small"

BATCH_SIZE = 1
MAX_CONTEXT_LENGTH = 128  # 减小上下文长度
